package library.client.domain.dto;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.Collections;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 
 *
 */
@Data
@ApiModel(description = "分页结果")
public class PageDTO<T> {
    @ApiModelProperty("总条数")
    private Long total;
    @ApiModelProperty("总页数")
    private Long pages;
    @ApiModelProperty("集合")
    private List<T> list;

    public static <PO, VO> PageDTO<VO> of(Page<PO> page, Function<PO, VO> converter) {
        PageDTO<VO> pageDTO = new PageDTO<>();
        pageDTO.setPages(page.getPages());
        pageDTO.setTotal(page.getTotal());
        List<VO> vos = Collections.emptyList();
        if (CollUtil.isNotEmpty(page.getRecords())) {
            vos = page.getRecords().stream().map(converter).collect(Collectors.toList());
        }
        pageDTO.setList(vos);
        return pageDTO;
    }

    public static <PO, VO> PageDTO<VO> of(Page<PO> page, Class<VO> voClass) {
        return of(page, (po) -> BeanUtil.copyProperties(po, voClass));
    }

    public static <VO> PageDTO<VO> of(Page<VO> page) {
        PageDTO<VO> pageDTO = new PageDTO<>();
        pageDTO.setPages(page.getPages());
        pageDTO.setTotal(page.getTotal());
        pageDTO.setList(page.getRecords());
        return pageDTO;
    }
}
